Function inside function in python